package com.ssm.entity;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class Commons {

    public final static List<String> TYPE_NAMES =  new ArrayList<String>(){{
        add("image");
        add("word");
        add("music");
        add("video");
        add("zip");
        add("bt");
        add("other");
    }};

    public final static Map<String,List<String>> TYPES = new HashMap<String, List<String>>(){{
        put("image",new ArrayList<String>(){{
            add("jpg");
            add("gif");
            add("png");
            add("ico");
        }});
        put("word",new ArrayList<String>(){{
            add("doc");
            add("docx");
            add("xml");
            add("txt");
            add("ppt");
            add("pptx");
            add("xls");
            add("xlsx");
        }});
        put("music",new ArrayList<String>(){{
            add("mp3");
        }});
        put("video",new ArrayList<String>(){{
            add("mp4");
            add("avi");
            add("wmn");
            add("mkv");
        }});
        put("zip",new ArrayList<String>(){{
            add("zip");
            add("rar");
            add("7z");
        }});
        put("bt",new ArrayList<String>(){{
            add("bt");
        }});
    }};

    public static String getTypeName(String suffix){
        for(String key:TYPES.keySet()){
            List<String> list = TYPES.get(key);
            if(list.contains(suffix)){
                return key;
            }
        }
        return "other";
    }

    public static void main(String[] args) {
        System.out.println(Commons.getTypeName("avi"));
    }
}
